package com.gang.util;
import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.geom.AffineTransform;
import java.awt.image.AffineTransformOp;
import java.awt.image.BufferedImage;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Date;

import javax.imageio.ImageIO;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.gang.model.Constant;
import com.gang.model.WAttachment;

/**
 * 图片工具类， 图片水印，文字水印，缩放，补白等
 * @author Carl He
 */
public final class ImageUtils {
    /**图片格式：JPG*/
    private static final String PICTRUE_FORMATE_JPG = "jpg";
    private static final Logger LOG = LoggerFactory.getLogger(ImageUtils.class); 
    private ImageUtils(){}
    /**
     * 添加图片水印
     * @param targetImg 目标图片路径，如：C://myPictrue//1.jpg
     * @param waterImg  水印图片路径，如：C://myPictrue//logo.png
     * @param x 水印图片距离目标图片左侧的偏移量，如果x<0, 则在正中间
     * @param y 水印图片距离目标图片上侧的偏移量，如果y<0, 则在正中间
     * @param alpha 透明度(0.0 -- 1.0, 0.0为完全透明，1.0为完全不透明)
*/
    public final static void pressImage(String targetImg, String waterImg, int x, int y, float alpha) {
            try {
                File file = new File(targetImg);
                Image image = ImageIO.read(file);
                int width = image.getWidth(null);
                int height = image.getHeight(null);
                BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
                Graphics2D g = bufferedImage.createGraphics();
                g.drawImage(image, 0, 0, width, height, null);
            
                Image waterImage = ImageIO.read(new File(waterImg));    // 水印文件
                int width_1 = waterImage.getWidth(null);
                int height_1 = waterImage.getHeight(null);
                g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));
                
                int widthDiff = width - width_1;
                int heightDiff = height - height_1;
                if(x < 0){
                    x = widthDiff;
                }else if(x > widthDiff){
                    x = widthDiff;
                }
                if(y < 0){
                    y = heightDiff;
                }else if(y > heightDiff){
                    y = heightDiff;
                }
                g.drawImage(waterImage, x, y, width_1, height_1, null); // 水印文件结束
                g.dispose();
                ImageIO.write(bufferedImage, PICTRUE_FORMATE_JPG, file);
            } catch (IOException e) {
                e.printStackTrace();
            }
    }
    /**
     * 添加文字水印
     * @param targetImg 目标图片路径，如：C://myPictrue//1.jpg
     * @param pressText 水印文字， 如：中国证券网
     * @param fontName 字体名称，    如：宋体
     * @param fontStyle 字体样式，如：粗体和斜体(Font.BOLD|Font.ITALIC)
     * @param fontSize 字体大小，单位为像素
     * @param color 字体颜色
     * @param x 水印文字距离目标图片左侧的偏移量，如果x<0, 则在正中间
     * @param y 水印文字距离目标图片上侧的偏移量，如果y<0, 则在正中间
     * @param alpha 透明度(0.0 -- 1.0, 0.0为完全透明，1.0为完全不透明)
     * @param weizhi 分别是图片的左上角,右上角,中间,左下角,右下角
*/
    public static void pressText(String targetImg, String pressText, String fontName
    		, int fontStyle, int fontSize, Color color, int x, int y, float alpha,int weizhi) {
        try {
            File file = new File(targetImg);
            
            Image image = ImageIO.read(file);
            int width = image.getWidth(null);
            int height = image.getHeight(null);
            BufferedImage bufferedImage = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
            Graphics2D g = bufferedImage.createGraphics();
            g.drawImage(image, 0, 0, width, height,null);
            g.setFont(new Font(fontName, fontStyle, fontSize));
            g.setColor(color);
            g.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_ATOP, alpha));
            
            int width_1 = fontSize * getLength(pressText);
            int height_1 = fontSize;
            int widthDiff = width - width_1;
            int heightDiff = height - height_1;
            if(x < 0){
                x = widthDiff;
            }else if(x > widthDiff){
                x = widthDiff;
            }
            if(y < 0){
                y = heightDiff;
            }else if(y > heightDiff){
                y = heightDiff;
            }
              if (weizhi == 1) {
                g.drawString(pressText, fontSize, fontSize);
               }else  if (weizhi == 2) {
                g.drawString(pressText, fontSize, height - fontSize);
               }else  if (weizhi == 3) {
                g.drawString(pressText, width / 2 - 2 * fontSize, height / 2);
               }else  if (weizhi == 4) {
                int sa = (pressText.getBytes().length / 2) * fontSize;
                g.drawString(pressText, width - sa, fontSize);
               }else if (weizhi == 5) {
                int sa = (pressText.getBytes().length / 2) * fontSize;
                g.drawString(pressText, width - sa, height - fontSize);
               }else{
            	   g.drawString(pressText, x, y + height_1);
               }
            
            g.dispose();
            ImageIO.write(bufferedImage, PICTRUE_FORMATE_JPG, file);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
    /**
     * 获取字符长度，一个汉字作为 1 个字符, 一个英文字母作为 0.5 个字符
     * @param text
     * @return 字符长度，如：text="中国",返回 2；text="test",返回 2；text="中国ABC",返回 4.
*/
    public static int getLength(String text) {
        int textLength = text.length();
        int length = textLength;
        for (int i = 0; i < textLength; i++) {
            if (String.valueOf(text.charAt(i)).getBytes().length > 1) {
                length++;
            }
        }
        return (length % 2 == 0) ? length / 2 : length / 2 + 1;
    }

    /**
     * 图片缩放
     * @param filePath 图片路径
     * @param height 高度
     * @param width 宽度
     * @param bb 比例不对时是否需要补白
*/
    public static void resize(String filePath, int height, int width, boolean bb) {
        try {
            double ratio = 0; //缩放比例    
            File f = new File(filePath);   
            BufferedImage bi = ImageIO.read(f);   
            Image itemp = bi.getScaledInstance(width, height, BufferedImage.SCALE_SMOOTH);   
            //计算比例   
            if ((bi.getHeight() > height) || (bi.getWidth() > width)) {   
                if (bi.getHeight() > bi.getWidth()) {   
                    ratio = (new Integer(height)).doubleValue() / bi.getHeight();   
                } else {   
                    ratio = (new Integer(width)).doubleValue() / bi.getWidth();   
                }   
                AffineTransformOp op = new AffineTransformOp(AffineTransform.getScaleInstance(ratio, ratio), null);   
                itemp = op.filter(bi, null);   
            }   
            if (bb) {   
                BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);   
                Graphics2D g = image.createGraphics();   
                g.setColor(Color.white);   
                g.fillRect(0, 0, width, height);   
                if (width == itemp.getWidth(null))   
                    g.drawImage(itemp, 0, (height - itemp.getHeight(null)) / 2, itemp.getWidth(null), itemp.getHeight(null), Color.white, null);   
                else  
                    g.drawImage(itemp, (width - itemp.getWidth(null)) / 2, 0, itemp.getWidth(null), itemp.getHeight(null), Color.white, null);   
                g.dispose();   
                itemp = image;   
            }
            ImageIO.write((BufferedImage) itemp, "jpg", f);   
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    public static WAttachment downPic(String imageUrl){
    	WAttachment t=new WAttachment();
		URL url;
		try {
			if(imageUrl.contains("upload")){
				
				url = new URL(imageUrl);
				//打开网络输入流
			    DataInputStream dis = new DataInputStream(url.openStream());
			    String[] fileNameStr=imageUrl.replace(Constant.wikiSite, "").split("/");
			    String pfileStr="";
			    for (int i = 0; i < fileNameStr.length-1; i++) {
			    	if(!"".equals(fileNameStr[i].trim())){
			    		pfileStr+=fileNameStr[i]+"/";
			    	}
				}
			    File pfile=new File(Constant.picFolder+pfileStr);
			    pfile.mkdirs();
			    String fName=fileNameStr[fileNameStr.length-1];
			    String newImageName=Constant.picFolder+pfileStr+fName;
			    //建立一个新的文件
			    LOG.info("文件地址"+newImageName);
			    File newFile=new File(newImageName);
			    
			    FileOutputStream fos = new FileOutputStream(newFile);
			    byte[] buffer = new byte[1024];
			    int length;
			    //开始填充数据
			    while((length = dis.read(buffer))>0){
			    	fos.write(buffer,0,length);
			    }
			    dis.close();
			    fos.close();
			    //加水印
			    pressImage(newImageName, Constant.water_pic, -1, -1, 0.5f);
			    if(imageUrl.contains(Constant.smallPicLabel)){
			    	downPic(imageUrl.replaceFirst(Constant.smallPicLabel, ""));
			    }
			    t.setAttachment(pfileStr+fName);
			    t.setCoindown(0);
			    t.setDownloads(0);
			    t.setFilename(fName);
			    t.setFilesize(newFile.length());
			    t.setFiletype(fName.split("\\.")[1]);
			    t.setFocus(0);
			    t.setIsimage(1);
			    t.setState(0);
			    t.setTime(new Date().getTime()/1000);
			}
			
		} catch (MalformedURLException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return t;
	    
	}
    public static void main(String[] args) throws IOException {
        pressImage("C://18.jpg", Constant.water_pic, -1, -1, 0.5f);
//        pressText("C://14.jpg", "旺仔之印 www.baidu.com", "宋体", Font.BOLD|Font.ITALIC, 20, Color.BLACK, -1, -1,1f,5);
//        resize("C://pic//4.jpg", 1000, 500, true);
    }
}